package com.dc.juc.example._1_count;

import com.google.common.base.Throwables;
import lombok.extern.slf4j.Slf4j;

import javax.annotation.concurrent.NotThreadSafe;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;

/**
 * Description: 多线程场景下，对一个普通变量进行 ‘++’ 操作，无法保证线程安全
 * @author: gdc
 * @date: 2021/7/28
 * @version 1.0
 */
@NotThreadSafe
@Slf4j
public class Count1 {

    /**
     * 请求总数
     */
    public static int CLIENT_TOTAL = 5000;
    /**
     * 同时并发执行的线程数
     */
    public static int THREAD_TOTAL = 200;

    /**
     * 累计变量
     */
    public static int COUNT = 0;

    public static void main(String[] args) throws InterruptedException {
        // newCachedThreadPool() 创建一个可变线程池
        ExecutorService threadPool = Executors.newCachedThreadPool();

        // Semaphore信号量，用来控制访问特定资源的线程数量，当前限制值为200
        Semaphore semaphore = new Semaphore(THREAD_TOTAL);

        // countDownLatch是一个计数器，线程完成一个记录一个，计数器递减，只能用一次
        CountDownLatch countDownLatch = new CountDownLatch(CLIENT_TOTAL);


        for (int i = 0; i < CLIENT_TOTAL; i++) {
            threadPool.execute(() -> {
                try {
                    // 获取一个令牌，在获取到令牌、或者被其他线程调用中断之前线程一直处于阻塞状态。
                    semaphore.acquire();

                    add();

                    // 释放一个令牌，唤醒一个获取令牌不成功的阻塞线程。
                    semaphore.release();
                } catch (InterruptedException e) {
                    log.error("异常信息为：{}", Throwables.getStackTraceAsString(e));
                } finally {
                    // 递减锁存器的计数，如果计数到达零，则释放所有等待的线程
                    countDownLatch.countDown();
                }
            });
        }

        // 使当前线程在锁存器倒计数至零之前一直等待
        countDownLatch.await();

        // 线程池关闭
        threadPool.shutdown();
        log.info("count:{}", COUNT);
    }

    /**
     * COUNT 计数 +1
     */
    private static void add() {
        COUNT++;
    }
}
